# coding: utf-8

import os
from selenium import webdriver
import time

chrome_options = webdriver.ChromeOptions()
# 使用headless无界面浏览器模式
chrome_options.add_argument('--headless')  # 增加无界面选项
chrome_options.add_argument('--disable-gpu')  # 如果不加这个选项，有时定位会出现问题

# 启动浏览器，获取网页源代码
browser = webdriver.Chrome(executable_path=r'/home/chan/IdeaProjects/study_daily/spider/chromedriver',
                           chrome_options=chrome_options)

epub_dir = '/media/chan/软件/LTLDataset/豆瓣影评/epubs/'


def get_page_urls(main_url):
    """
    获取首页中电影地址
    :param main_url:
    :return:
    """
    browser.get(main_url)  ##主页
    time.sleep(3)
    epub_names = os.listdir(epub_dir)
    for i, epub in enumerate(epub_names):
        if i % 9 == 0:#10个文件就要刷新一下，点击选择文件
            browser.find_element_by_xpath('/html/body/div[2]/div[1]/div[1]').click()
        epub_path = os.path.join(epub_dir, epub)
        browser.find_element_by_xpath('//*[@id="upload_0"]').send_keys(epub_path)
        time.sleep(0.02)
        print(i)


if __name__ == "__main__":
    main_url = r'http://10.35.21.92:12121/'#WIFI传书地址
    get_page_urls(main_url)
